import csv
import sqlite3

connection = sqlite3.connect('../db.sqlite3')
cursor = connection.cursor()

create_table = '''
create table data_SSD
(
    Name text not null,
    ID text not null,
    PID text not null,
    Platform text not null,
    Price text,
    Performance text,
    Capacity text,
    Slot text,
    PCIESlot text,
    ReadSpeed text,
    WriteSpeed text,
    ReadSpeed4K text,
    WriteSpeed4K text,
    Type text,
    constraint ssd_pk
        primary key (Name,ID,PID,Platform)
);

'''

cursor.execute(create_table)
file = open('../GetInfo/Details/SSDInfo.csv', encoding='utf-8')
contents = csv.reader(file)
insert_records = "INSERT INTO data_SSD (Name,ID,PID,Platform,Price,Performance,Capacity,Slot,PCIESlot,ReadSpeed,WriteSpeed,ReadSpeed4K,WriteSpeed4K,Type) " \
                 "VALUES(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)"

cursor.executemany(insert_records, contents)
select_all = "SELECT * FROM data_SSD"
rows = cursor.execute(select_all).fetchall()
for r in rows:
    print(r)

# Committing the changes
connection.commit()
connection.close()
